Added basic upgrade support to begin balancing mod