Added ability to pass in an image as an argument. No longer hard coded.