Forgot to commit this before. Gonna code some restrictions...