creating separate classes
Merge https://github.com/daajdk/hw4