Small changes before gym test