added more functionality, updated readme