Cleaned up homework01a a bit