[DAY 2] working on second class