Refactored + added getScreenSize() feature