Allow entity() to take in String class name