Modified code and organized photos into folder