Complete the code for the client. Going to test later.