Refactor push notification classes.