Todo: fix first feature