Updated a bug. Just wanted to get it working for the night.