Solved exercise 1.2, gonna add comments so I remember what I did wrong
the first time.