minor code refectoring #22

Closed mshariq-nerd closed 7 years ago