Outsource stages into enum (Stage change not working yet)