Added small future work section
Added small future work section