Added remaining images and updated code accordingly. Resolves #1