adding code for automated test