Cleaned up a bit

Signed-off-by: Jeroen van Baarsen <jeroenvanbaarsen@gmail.com>