Update class.cpp to avoid else after return

Update class.cpp to avoid else after return #734