Fixing html to reflect changes to the model