Added note that the code is only tested with python 2.7