Added argument parsing to accept user input for number of iterations
Added separate directories for data and images
Added gitignore
Added README