Start adding functionality and input fields to generate images