Add some tests
Update patch