UI done but need to consider background worker