Editted and changed checkpoint 2
Down sampler implemented