Changed program structure:

*Include class from header file.
*TODO: Bugfixes