If the author is here, this is perfect except for 'space' not being able to click a button