fixing some code...