Add future work section