add remove unit tests