Added future work comments