Added more changes from RTS Demo

Updated files to include more code from RTS demo.