Remove unused class and change argument name from 'json' to 'return_json'