To make it vegan, I would recommend using unsweetened non-dairy milk and adding nutritional yeast to the sauce instead of the shredded cheese. If you know of a good quality vegan cheese that melts well, you can also swap that in.