Added some stuff to future work