added arjun's code