Refactor. Added another simple game