added functionality

yay