Improve regex