Update readme and use log instead of print