Almost everything done but code is fucked up