First Trial for HW 2

I still need to fix print formatting and add comments.