Worked on Assignment 5 a bit