Added middleware back in using the new module