Updated and added comments for assignment