Fixed bug where only one class was showing up