Used __name__ instead of class_key