Small bug fix and readme update