Basic test code added