Separate code on another classes