Actually, my preference would be that if a filename is passed it refers to a PNG or other color image, not a 3-color FITS file. However, it should be possible to get both to work. If you're note sure how best to proceed I can open a PR on your branch.