updated functionality

 homework