Fix! Now draws two different box types