finish adding my code