added button and changed behaviour
notes
notes