first homework finished, started 2nd

Todo - refactor all problems