Forgot to add citation handling