Refactor and generate collage

- Refactor code
- Create a collage generator pop up