Commented test code, added stdin input