Tweak reward function to always penalise based on distance at episode termination