Move code into class