Please add unit tests for these methods that you are adding