added idea to future works
Wrote the Additional Improvements sections.