Update main.py

Added Think Python and A Byte of Vim