Add test, update regex