It should be very easy to extend the LensedPS class in order to account for a non-lensed PS. This will be useful for codes like MOLET and Herculens. I can add this, for example, by adding a "lensed/unlensed" flag to the class (because only a single image can still be there after lensing). Or we could include the unlensed source position and flux as extra but optional arguments to this class.
Of course when it comes to the plotting API, this will mean that we need our own COOLEST way to find the multiple images if only the unlensed source is given. Or not, we can leave this out, and the plots would check this flag and plot the point source accordingly, either on the lens and/or the source planes.