still trying to setup classes