fixed button press logic so can grab better now