Fix code for misunderstanding requirement. Should generate 480x480 image instead of the sample outputs which are 504x504.