Add get_policy functionality