Fix class name
Update build