Moved class into main

Moved class into main for submission to codeeval