Not sure exactly where to put the coding.
For it to work correctly.