Finished with first part on assignment 4