straight logic - new approach + buttons