ucb python seminar homework #1