Modified knights class and tested it