Still not getting the correct answer on CodeEval