Pulizia e bug fixes