Finished the entire assignment 5