Added more UI

Added more UI to main.py. Still a draft. Only GUI, no functionality.