Moving logic to player entity