This assignment is built using the code from assignment 5