Improvements in Imitation Learning for Overcooked