Converted to numpy

and added pygame version of the main loop