Fixed main bug
Fixed main bug