fixed mistakes from class