Dis code has problems