Wrote assignment part!

Fixed #2 :D