Added power-up logic