Fix #15, add end-of-shift button