Half way done with assignment 4