Add challenges and future work