My assignment submission

Improved the comments