Challenge 3 completed

TODO:
    - Use readline
    - Check input