Expand on Future work section