Moved image loading to main func, created data folder and files, cleaned code