Just have to do comments and readme