Commit need the end of homework

Adding comments, refactoring