working on adding button