Minor changes to DOM