Add more code for adding/viewing image