Button Fix

Fixed button code with new button syntax.