changed some minor stuff in terms of plotting etc.