minor bug repair
first iteration done