Move class to samples package