add remove code + fix