just fixing some stuff !!