small changes to future work