Move client only module logic to the client folder