use input instead of raw_input (more python3)
Python 3 print() - remove python2 version