Fixed the issues pointed by Udacity staff and also added implementation for the suggestions I made for PRoblem #5 and #6