Fix UX of add button