Added Screen classes and changes in main class.