Made it work for multiple initial points and added colors