minor change.
Code is buggy