Add behavior
Refactor update for image