Made checkpoint location an env variable