Move code to src