Finish assignment 2!
Add logs and output