Separate class and define