Add code for number input and AddButton