Refactor code and delete unnecessary images and files