Added some extra functionality, try pressing e while the program is running