Fix the alignment of the button - make it appear next to the input