Little commit (moved some code around in Load_model).