Finished assignment 6