Updated the layout back to the way its was