trying to fix compiler error b4 class ended