Add missing module to setup.py
Move JSON class, change its name and fix imports.