Reset button added and UI improved