Changed some stuff around. For simplicity.