Adding src code for the first draft