Task Focused Robotic Imitation Learning