added my own code, updated the README file