Finalized assignment 4