Made Changes to updateLayout

Yep thatt's what I did.