Added ability to extra check images!

Also restructred writer class.