adding original code for testing