Small Fixes in homework