File seperation, class-based
Added in 2 different color choices