A couple more lines of code here and there.