fixing some code.