Added functionality to follow mouse