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

fixed the design issue