Fixed main to include image creation