modified cout statements and added more classes