Fix module name and extension so that it loads succesfully in python