added new functionality

good job!