problem solved! update readme and code