added functionality

close #1