added some code and fixed readme