train class