Updated heading and Challenges and Future Work section