Edited submission

Fixed it to better meet the requested program (only have 1 class that
you can be good in)