Attempt to pass first task

Hello! 

The code is already quite clean and it's working. Check it out