Fixed bugs v2