touching things up
additional function added