Updated UI with edit text field and working button