Add server side code as well