Added basic ws code