Moved UnitCollection creation into a method, made code more readable, fixed threat level not being calcuated at the right point