BABYAI Learning to pump (BABYAI)