Finished graphical implementation using pygame