code for STS challenge