fixed some design issue
fixed the design issue
Merge pull request #173 from ritesharora89/mybranch

Mybranch