Screen as Class, added some error handling