Problem Space Transformations for Generalisation in Behavioural Cloning