Restore deleted functionality