button fixes
Button fixes