changed code to test