load_state basically working
loading state works, testing is now quicker.