Little code errors made