Finished assignment 4