Fix qualiyty flaws