worked on 'Fixed the UI changes'