Code for codeeval