Cleanup

Signed-off-by: Samuel Ward <samuelpward@gmail.com>