Add ability to run build on test

I've added the ability to run the build on test as well as deploy and tidied a few bits up that I came across.

Open for discussion.
Maybe change it to: 'On which phase would you like to build your image?'