Almost finished assingment 2 on homework 2