Update __main__.py

added support for arrows keys