CodeEval challenge number 004