Adding classes for week 04 assignment
Solution for week04 assignment