added handling for local classes