still working on making classes right