I think you need one more variable in there, but yeah that's the right idea :)