added gui.py, modified main.py little making it a function(ask anurag for it)