Add Future Work section