Looks like I still need this code, unless I'm missing something.