Move code to class