I liked this one. I would recommend adding some kind of delay to the input in the Upgrade your unit screen; there were a few cases where I was spamming the little guys and accidentally, upgraded them instead of another unit. I would also recommend moving the upgrade options a little higher to prevent spam clicking. Finally, add a timer to indicate "Upgrade in x seconds"