Some small changes. Added a button in GUI.