Very minor code improvement, updated readme