Added a pygame based UI