fixed readme and added new code