Completed assignment

Added more comments