Minor refactor 

Added category on SKNode and refactored some code