updates

touch ups to the code