Few corrections to assignmnet 2