Code changes after class