GUI Code

Modify look of GUI and modify class