Added main for demo purposes