-Adding validation for size, channels and colorspace -Refactoring to throw general exceptions -Creating tests -Using other better data types