fixed bug
fixed main bug