worked on adding listener for click