add ans class, problem not solved