Created interface class
Added implementation
Changed main function
Example of the output