Finished seamingly working version of Assignment 01