Fixed the button container a bit