BUG: Allow integer for image dimension
BUG: Fixed image dimension for output images