Add omit method for functional and integration tests