Now this class can work