Major code work - Check class basically works but still need to do some stuff on Code.java