Fixed setup.py for compiling extension