IMP: jsut change class name and define wheather