Changed model to take as input and outout relative position