Added back in the saving feature in Learn.py and changed some configs