Improved upgrade animation