Update HW5

Added button handle