Fix Code

[brebenelmihnea]

Place an "x" in the corresponding checkbox if it is done or does not apply to this pull request.

• [x] I have tested my code.
• [x] I have added my solution according to the steps here.
• [x] I have followed the code conventions mentioned here.
  • I understand that if it is clear that I have not attempted to follow these conventions, my PR will be closed.
  • If changes are requested, I will re-request a review after addressing them.
• [ ] I have linked this PR to any issues that it closes.