A simple way to incorporate loss when modelling multimode entangled state generation