Fixed some bugs cleaned up some code. Works great now.
Also added a prompt which screen should be captured.
forgot this include