Add onclose handler
Update demo