Almost finish assignment 5
Assignment 5