Fixed bug in FitsImage class in notebook - now generative model is sensible