added more functionality and updated README