Code

updated slightly