add button code moved to enter button