Initial attempt on assignment 1