code seems to be working jajaj