Rearrange step function to eat food correctly
Minor refactoring