Added key scheduling code