Add task class and sample_task class