Update code for server