Comment by bestcoder

I went with your suggestion but changed a little I think I made it worse