Finished hw1-2. Need to get familiar with python features.