Updated the UI. It should be more user friendly now.