Added a main() function for quick testing