Transformers and Slot Encoding for Sample Efficient Physical World Modelling