Finish assignment

Might want to add some comments later.