Finished assignment 5