Finished assignment

I learnt a lot from this!