Added Problem 3
Update draw_states.m for Problem 1