Add states to animation screen