Updating Layout code