Improve register use tracking