1st attempt to Assignment 2