updated div class to keep it within container