change class, move to main