Updating layout according to new design