Added .zip functionality and modified main()