Add basic button code