First attempt at assignemnt 2