Homework assignment 2

First try at the programming assignment