added test to StevenYiu file