improved code

Authored by anmolgautam on May 28 2018, 5:24 PM.