Add get_status function