Tried adding my code, not quite working