Add comments and fix readme