Added working main for testing