making it py2.x compatible (cause I wasn't able to get run 3.x matplotlib on Linux Mint)