Update program to accept image as an argument