Thanks for the comment, I unfortunately didnt have enough time, it was supposed to work like you suggested were you select the upgrade you want but it was bugged for some reason.
