Added classes for second homework