made some small additions that were necessary