NEWFEATURE: SpaceClassification works with mse loss too