Rewrite - create a separate Photo class