Assignment 5 correction done