Clean up WIP functions

Signed-off-by: Zeklandia <awood5659@gmail.com>