Homework-4, improved.