Edited heading and added a div class