updates to Cole's code
updates to Cole's code