Imported from https://github.com/NetraInamdar/Deep_Learning_Project